package com.im.common.netty.upload.server;

import com.im.common.netty.upload.codec.UploadFileDecoder;
import com.im.common.netty.upload.handler.UploadFileHandler;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;

/**
 * 文件上传服务器
 */
public class UploadFileServer {

    private int port;

    public UploadFileServer(int port) {
        this.port = port;
    }

    public void run() throws Exception {
        EventLoopGroup bossGroup = new NioEventLoopGroup(1); // 处理客户端请求的 boss 线程组
        EventLoopGroup workerGroup = new NioEventLoopGroup(); // 处理接收到的连接的 worker 线程组

        try {
            // 创建服务器的引导对象
            ServerBootstrap b = new ServerBootstrap(); // (2)
            // 将 bossGroup 和 workerGroup 配置到 ServerBootstrap 中，在 boss 线程组中，每个 EventLoop 会监听一个端口，等待客户端的连接请求。
            // 一旦接收到连接请求，boss 线程组会将连接分配给一个 worker 线程组中的 EventLoop 进行处理
            b.group(bossGroup, workerGroup)
                     // 使用 NIO 的方式处理 I/O 操作
                    .channel(NioServerSocketChannel.class) // (3)
                    .childHandler(new ChannelInitializer<SocketChannel>() { // (4)
                        @Override
                        public void initChannel(SocketChannel ch) throws Exception {
                            ChannelPipeline pipeline = ch.pipeline();
                            pipeline.addLast(new UploadFileDecoder());
                            pipeline.addLast(new UploadFileHandler());
                        }
                    })
                    .option(ChannelOption.SO_BACKLOG, 128)          // (5)
                    .childOption(ChannelOption.SO_KEEPALIVE, true); // (6)

            // 绑定并开始接受传入连接
            System.out.println("tcp start success");

            // 绑定端口，启动服务
            // 使用 bind() 方法绑定端口，然后使用 sync() 方法等待绑定完成
            ChannelFuture f = b.bind(port).sync(); // (7)

            // 等待 Channel 关闭
            f.channel().closeFuture().sync();
        } finally {
            // 释放资源
            workerGroup.shutdownGracefully();
            bossGroup.shutdownGracefully();
        }
    }

}
